﻿
package wesi.Table;

import java.awt.Component;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Rectangle;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import java.util.Iterator;

import javax.swing.JComponent;
import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.UIManager;
import javax.swing.plaf.basic.BasicTableHeaderUI;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableColumn;
import javax.swing.table.TableColumnModel;

/**
 * Created by IntelliJ IDEA.
 * User: nastya.kuz
 * Date: 29.07.2010
 * Time: 17:35:20
 * <p/>
 * Модифицирование стандартного класса javax.swing.plaf.basic.BasicTableHeaderUI
 * с учетом того, что теперь работа происходит со сложным многоуровневым заголовком.
 */
public class GroupableTableHeaderUI extends BasicTableHeaderUI {

	/**
	 * @param graphics  the Graphics context in which to paint
	 * @param component the component being painted; this argument is often ignored, but might be used
	 *                  if the UI object is stateless and shared by multiple components
	 */
	public void paint(Graphics graphics, JComponent component) {
		Rectangle clipBounds = graphics.getClipBounds();
		if (header.getColumnModel() == null) {
			return;
		}
		((GroupableTableHeader) header).setColumnMargin();
		int columnCount = 0;
		Dimension size = header.getSize();
		
		Rectangle cellRectangle = new Rectangle(0, 0, size.width,size.height);
		Hashtable<ColumnGroup, Rectangle> hashTable = new Hashtable<ColumnGroup, Rectangle>();
		
		Enumeration<TableColumn> enumeration = header.getColumnModel().getColumns();
		while (enumeration.hasMoreElements()) {
			cellRectangle.height = size.height;
			cellRectangle.y = 0;
			TableColumn aColumn = enumeration.nextElement();
			
			@SuppressWarnings("rawtypes")
			Iterator columnGroups = ((GroupableTableHeader) header).getColumnGroups(aColumn);
			if (columnGroups != null) {
				int groupHeight = 0;
				while (columnGroups.hasNext()) {
					ColumnGroup columnGroup = (ColumnGroup) columnGroups.next();
					Rectangle groupRectangle = hashTable.get(columnGroup);
					if (groupRectangle == null) {
						groupRectangle = new Rectangle(cellRectangle);
						Dimension d = columnGroup.getSize(header.getTable());
						groupRectangle.width = d.width; // TODO d.width-2;
						groupRectangle.height = d.height;
						hashTable.put(columnGroup, groupRectangle);
					}
					paintCell(graphics, groupRectangle, columnGroup);
					groupHeight += groupRectangle.height;
					cellRectangle.height = size.height - groupHeight;
					cellRectangle.y = groupHeight;
				}
			}
			cellRectangle.width = aColumn.getWidth();
			if (cellRectangle.intersects(clipBounds)) {
				paintCell(graphics, cellRectangle, columnCount);
			}
			cellRectangle.x += cellRectangle.width;
			columnCount++;
		}
	}

	/**
	 * отрисовывает столбец в таблице
	 *
	 * @param graphics    графический контекст в котором происходит прорисовка
	 * @param cellRect    размеры столбца
	 * @param columnIndex индекс столбца
	 */
	private void paintCell(Graphics graphics, Rectangle cellRect,int columnIndex){
		TableColumn aColumn = header.getColumnModel().getColumn(columnIndex);
		TableCellRenderer renderer = new DefaultTableCellRenderer() {
			private static final long serialVersionUID = 1L;

			public Component getTableCellRendererComponent(
					JTable table, Object value, boolean isSelected,
					boolean hasFocus, int row, int column) {
				JLabel header = new JLabel();
				header.setForeground(table.getTableHeader().getForeground());
				header.setBackground(table.getTableHeader().getBackground());
				header.setFont(table.getTableHeader().getFont());
				header.setHorizontalAlignment(JLabel.CENTER);
				header.setText(value.toString());
				header.setBorder(UIManager.getBorder("TableHeader.cellBorder"));
				return header;
			}
		};
		Component component = renderer.getTableCellRendererComponent(
				header.getTable(), aColumn.getHeaderValue(), false,
				false, -1, columnIndex);
		component.setBackground(UIManager.getColor("control"));
		rendererPane.add(component);
		rendererPane.paintComponent(graphics, component, header,
									cellRect.x, cellRect.y, cellRect.width,
									cellRect.height, true);
	}

	/**
	 * отрисовывает группу столбцов в таблице
	 *
	 * @param graphics графический контекст в котором происходит прорисовка
	 * @param cellRect размеры группы
	 * @param cGroup   индекс столбца
	 */
	private void paintCell(Graphics graphics, Rectangle cellRect,
			ColumnGroup cGroup) {
		TableCellRenderer renderer = cGroup.getHeaderRenderer();
		Component component = renderer.getTableCellRendererComponent(
				header.getTable(), cGroup.getHeaderValue(), false,
				false, -1, -1);
		rendererPane.add(component);
		rendererPane.paintComponent(graphics, component, header,
				cellRect.x, cellRect.y, cellRect.width,
				cellRect.height, true);
	}

	/**
	 * вычисляет (суммарную по столбцам?) высоту header
	 *
	 * @return высота
	 */
	private int getHeaderHeight() {
		int height = 0;
		TableColumnModel columnModel = header.getColumnModel();
		for (int column = 0; column < columnModel.getColumnCount(); column++) {
			TableColumn aColumn = columnModel.getColumn(column);
			TableCellRenderer renderer = aColumn.getHeaderRenderer();
			
			if (renderer == null) {
				// высота по умолчанию
				return 33;
			}
			Component component = renderer.getTableCellRendererComponent(header.getTable(),aColumn.getHeaderValue(), false, false, -1,	column);
			int columnHeight = component.getPreferredSize().height;
			@SuppressWarnings("rawtypes")
			Iterator iterator = ((ArrayList) ((GroupableTableHeader) header).getColumnGroups(aColumn)).iterator();
			if (iterator != null) {
				while (iterator.hasNext()) {
					ColumnGroup cGroup = (ColumnGroup) iterator.next();
					columnHeight += cGroup.getSize(header.getTable()).height;
				}
			}
			height = Math.max(height, columnHeight);
		}
		return height;
	}

	/**
	 * вычисляет размеры header исходя из предпочтительной ширины
	 *
	 * @param width ширина
	 *
	 * @return размеры (ширина, высота)
	 */
	private Dimension createHeaderSize(long width) {
		TableColumnModel columnModel = header.getColumnModel();
		width += columnModel.getColumnMargin()
		* columnModel.getColumnCount();
		width = Math.min(width, Integer.MAX_VALUE);
		return new Dimension((int) width, getHeaderHeight());
	}

	/**
	 * возвращает предпочтительный размер для header
	 *
	 * @param component компонент, для которого вычисляется размер (в функции не используется)
	 *
	 * @return возвращает предпочтительный размер компонента
	 */
	public Dimension getPreferredSize(JComponent component) {
		long width = 0;
		Enumeration<TableColumn> columns = header.getColumnModel().getColumns();
		while (columns.hasMoreElements()) {
			TableColumn aColumn = (TableColumn) columns.nextElement();
			width = width + aColumn.getPreferredWidth();
		}
		return createHeaderSize(width);
	}
}
